package com.shenzhen.horselive.widget.zxing.encode;

import android.graphics.Bitmap;
import android.graphics.Canvas;
import android.text.TextUtils;

import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.WriterException;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.qrcode.QRCodeWriter;

import java.util.Hashtable;


public class CodeCreator {

	/**
	 * 生成QRCode（二维码）
	 * 
	 * @param str
	 * @return
	 * @throws WriterException
	 */
	public static Bitmap createQRCode(String url) throws WriterException {

		if (url == null || url.equals("")) {
			return null;
		}

		// 生成二维矩阵,编码时指定大小,不要生成了图片以后再进行缩放,这样会模糊导致识别失败
		BitMatrix matrix = new MultiFormatWriter().encode(url,
				BarcodeFormat.QR_CODE, 300, 300);

		int width = matrix.getWidth();
		int height = matrix.getHeight();

		// 二维矩阵转为一维像素数组,也就是一直横着排了
		int[] pixels = new int[width * height];

		for (int y = 0; y < height; y++) {
			for (int x = 0; x < width; x++) {
				if (matrix.get(x, y)) {
					pixels[y * width + x] = 0xff000000;
				}

			}
		}

		Bitmap bitmap = Bitmap.createBitmap(width, height,
				Bitmap.Config.ARGB_8888);
		bitmap.setPixels(pixels, 0, width, 0, 0, width, height);
		return bitmap;
	}


	/**
	 *
	 * @param content                字符串内容
	 * @param width                  二维码宽度
	 * @param height                 二维码高度
	 * @param character_set          编码方式（一般使用UTF-8）
	 * @param error_correction_level 容错率 L：7% M：15% Q：25% H：35%
	 * @param margin                 空白边距（二维码与边框的空白区域）
	 * @param color_black            黑色色块
	 * @param color_white            白色色块
	 * @param logoBitmap             logo图片
	 * @param logoPercent            logo所占百分比
	 * @return
	 */
	public static Bitmap createQRCodeBitmap(String content, int width, int height, String character_set,
                                            String error_correction_level, String margin, int color_black,
                                            int color_white, Bitmap logoBitmap, float logoPercent) {
		// 字符串内容判空
		if (TextUtils.isEmpty(content)) {
			return null;
		}
		// 宽和高>=0
		if (width < 0 || height < 0) {
			return null;
		}
		try {

			/** 1.设置二维码相关配置,生成BitMatrix(位矩阵)对象 */
			Hashtable<EncodeHintType, String> hints = new Hashtable<EncodeHintType, String>();
			// 字符转码格式设置
			if (!TextUtils.isEmpty(character_set)) {
				hints.put(EncodeHintType.CHARACTER_SET, character_set);
			}
			// 容错率设置
			if (!TextUtils.isEmpty(error_correction_level)) {
				hints.put(EncodeHintType.ERROR_CORRECTION, error_correction_level);
			}
			// 空白边距设置
			if (!TextUtils.isEmpty(margin)) {
				hints.put(EncodeHintType.MARGIN, margin);
			}
			/** 2.将配置参数传入到QRCodeWriter的encode方法生成BitMatrix(位矩阵)对象 */
			BitMatrix bitMatrix = new QRCodeWriter().encode(content, BarcodeFormat.QR_CODE, width, height, hints);

			/** 3.创建像素数组,并根据BitMatrix(位矩阵)对象为数组元素赋颜色值 */
			int[] pixels = new int[width * height];
			for (int y = 0; y < height; y++) {
				for (int x = 0; x < width; x++) {
					//bitMatrix.get(x,y)方法返回true是黑色色块，false是白色色块
					if (bitMatrix.get(x, y)) {
						pixels[y * width + x] = color_black;//黑色色块像素设置
					} else {
						pixels[y * width + x] = color_white;// 白色色块像素设置
					}
				}
			}

			/** 4.创建Bitmap对象,根据像素数组设置Bitmap每个像素点的颜色值,并返回Bitmap对象 */
			Bitmap bitmap = Bitmap.createBitmap(width, height, Bitmap.Config.ARGB_8888);
			bitmap.setPixels(pixels, 0, width, 0, 0, width, height);

			/** 5.为二维码添加logo图标 */
			if(logoBitmap != null){
				return addLogo(bitmap, logoBitmap, logoPercent);
			}
			return bitmap;
		} catch (WriterException e) {
			e.printStackTrace();
			return null;
		}
	}

	/**
	 * 向二维码中间添加logo图片(图片合成)
	 *
	 * @param srcBitmap 原图片（生成的简单二维码图片）
	 * @param logoBitmap logo图片
	 * @param logoPercent 百分比 (用于调整logo图片在原图片中的显示大小, 取值范围[0,1] )
	 * @return
	 */
	private static Bitmap addLogo(Bitmap srcBitmap, Bitmap logoBitmap, float logoPercent){
		if(srcBitmap == null){
			return null;
		}
		if(logoBitmap == null){
			return srcBitmap;
		}
		//传值不合法时使用0.2F
		if(logoPercent < 0F || logoPercent > 1F){
			logoPercent = 0.2F;
		}

		/** 1. 获取原图片和Logo图片各自的宽、高值 */
		int srcWidth = srcBitmap.getWidth();
		int srcHeight = srcBitmap.getHeight();
		int logoWidth = logoBitmap.getWidth();
		int logoHeight = logoBitmap.getHeight();

		/** 2. 计算画布缩放的宽高比 */
		float scaleWidth = srcWidth * logoPercent / logoWidth;
		float scaleHeight = srcHeight * logoPercent / logoHeight;

		/** 3. 使用Canvas绘制,合成图片 */
		Bitmap bitmap = Bitmap.createBitmap(srcWidth, srcHeight, Bitmap.Config.ARGB_8888);
		Canvas canvas = new Canvas(bitmap);
		canvas.drawBitmap(srcBitmap, 0, 0, null);
		canvas.scale(scaleWidth, scaleHeight, srcWidth/2, srcHeight/2);
		canvas.drawBitmap(logoBitmap, srcWidth/2 - logoWidth/2, srcHeight/2 - logoHeight/2, null);

		return bitmap;
	}
}
